train_data_folder = 'data/train'
test_data_folder = 'data/val'
batch_size = 16
num_workers = 0
shuffle = True
epochs = 300
lr = 0.01
momentum = 0.95
validation_freq = 500